Cleaned up unit tests file, and added TODO's for HW6.