Big code tidy