Wrote unit tests for entity classes