Bug fixes
Fixed #3
Fixed #4