modified a.py
unified everything