adding unit tests for what methods should be included