exam 2 prep, need to add some code i wrote in notebook