Added classes to make gui work