Updated test fit script