hw3 - update solution from hw4 files