PR for #11

Fixes #11
done