Added in the functions, not sure if it'll work though.