A little rafactoring ;)