Reformatted code, added the __future__ import