Updated code

Updated code to reflect changes made in dependencies.