Add gui functionality