added basic tests for code