task01 code cleand up, all good