0.2.1: Nicer code and better output