Revised homework for edX submission