Yeah, I thought I should move this logic into a helper method. And again I have forgot to do it. Thanks :)