Add new features to the scripts