Add full python highlighting