Small formatting change and adopt __future__