Update atom-detection code