Adding test python file