Fix setup script

Closes #29