Revert "Revert "Added challenge.py""

This reverts commit dd620a89b1436cefa54fa615b4162ebac37bcff8.