Fixed display issue

Forgotten code was causing a bad height issue on first selection box,
removed it now.
Also moved some initiation code out of the main function to tidy it up a
bit.