Ok, sorry for the mistake if any. Still learning my way around there :( I'll make another pull request.