Add support for u-boot