Fix SVG bug & clean up code