updated code and added 2 vars

added two vars of type int