Add new funcitonality