Updated ATOM code
Merge branch 'main' of https://github.com/akshaybapat/ATOM.git