added missing functionality