HW1-4.1 - problem solved