What do you think about also handling atom inputs? I'm happy to make a PR to implement this enhancement if you're on board.