Fixing week-1 10 challenge