Switch to Sage

Made changes to reflect that this is for Sage, not Python (although the syntax is the same).