added way to get return value after running program