Refactored task into a class and added shuffle phase