Tidy up old code