Add better python highlighting