#35 fix bugs