updating assignment 2 files