It's a good idea to create a property for elements like buttons. 

```python
@property
def save_button(self):
    return self.get_element(*self._save_button)
```