Started working on the gui part