Copied old test and updated for current version

Closes #9