update code to pass specs