update functions from initial tests