Changing the code to handle variable work group sizes