- Code tidy ups