Tidy up cdoe