Fixed #19 and Fixed #20