Utility script to setup training assistant