tidyup code