adding new code to my new feature