[FEATURE] Added more functionality to the benchmark creation.