Reinstate logging

Few minor changes, fully removes use of `print_update`