How to create nested function in python