Basic UI improvements

Tidied up the elements. Replaced 'visibility: hidden' with 'display:
none' so that hidden elements don't leave empty space.

UI needs work.