Move this over to Kconfig and tidy up.

Signed-off-by: Simon Glass <sjg@chromium.org>