Fixed #19 works now