reworking week 2's assignment