added one more function to code