Assignment 2

similar to last time