add assignment 2
midify the bottom