Added some stuff to gui