Work on update functions

Added update function to account