update python code to reflect needing to print return value