Added check on workgroup size