Changed logger in order to make it more customisable