fix cls in py shell