Added class to handle updating atm