code tidie up