Add class python to all code sections