feature update
Fixes #4