updated more code