Wrote tests for existing classes