creating my own generator function
modified the add.py function to make it work as module