Working version
changed name of input variable