adding tests that i finally understand