python highlighting improvements