Library changed, changed code to reflect that