added additional functionality