Fixed issue #1 and updated readme

Close #1