Add back functionality to away button