Added some functionality that I forgot about ie. the get functions for the classes