code tidyup