added some home task on python