Added logger functionality