Tidied up some code
Made buttons smaller