tiny fix with the matrix stack managment in atomo.py
tiny clean up