fixing added code