Draft version of homework week 4