ToDo & tidy up