Update setup
Fixes #3