Added rainbow function to warlock