add 2 codewars.py