Finished assignment week 3