Cleanup to linq statement