Fixed errors

Fixes #2