added lol function in ex2a.py