Tidied up code. Should look nicer now on smaller screens especially.