Fix installation

Closes #9 
:thumbsup: