Add new funtionality