update code for new engine imporvements