Assignment 4 final submission