Added some styling - deleted unused code