Added some comment to the classes that arent quite working