New version 2.0.1

Fixed some bugs