updated code

Variable assigned to code