Fix #7: add imports and remove extra blank line