Add tests and finish grader.js