TIdy up
More tidy up