Updating the code to be UE4.4 compatible