Updated: Nvm, misread the silly diff. I see you change the method signature a little. Looks cleaner!