added features to program