fix issue #5 & #6