Implement driver script properly