Tidying up code.