added some code for the exam