Updates after comments

Signed-off-by: Joakim Bech <joakim.bech@linaro.org>