Adds a function to get atoms by type