New version 0.1.1, fixed a few bugs