Added some basic styling;

cleaned up code