added features to make this program more user friendly