Tidy up tests by removing unnecessary anonymous functions
Added end of update function