Re-add attention getter. We'll revisit it in #201