small fixes

I'll add a couple more later but nothing major. Cool project!