fixing code
more code