small layout changes in the code