Feature added. Fixes #3.