completed assignment - need to test before submission