Fix typo and imports #7