Some code I am writing while trying to learn vector math in Python