Add missing import, remove bad spaces

Part of #30