Added python test