Finished up week3 assignment