Adding additional functionalities