baseline code for assignment 2