some bugs should be fixed

issue #5