fixed to make incorrect problem when given 1