Updated logger a bit more