Tidy Up code.