Special relativity simulator for Python