Adding in logging, shitty for now