Added python script for tests