Added more classes

Added some more classes to manage different parts of the plugin