Tidy up nast code