Assignment 4: All tests passing