some small updates in main file and DOM.py