Cleaned up tests for add and check