Resolved #3

Fixing #3