Unit tests for the classes