Add lint and fix issues #16

What the title says.