Initial version of homework 5