Adding tests for the exam problems