updated code

added to variable of type int