Fixed previous commit, added one function call