Small improvements

add whitespace