cleaned code a little
variable name fix