Quick tidyup