deixei o codigo mais OO