Add more gui functionality