Fixes #1
Update for #1