Update ex21.py

la til en liten kodesnutt