DONE: Feature works after program reload