Created tests for the major classes