updated code

added to variables of type int