solution for Assignment 2
solution for Assignment 2 w/ test func