Adds basic test coverage