looks like it works now. verified it with puzzle 1 and 2