better tests for all 3 problems of assignment 4