make it works with code in multiple linesss