improved gui and added functionality