Working on pretifying the GUI