Add statements to let student know if correct or not