completed homework 3 - need to test