Use the log package instead of fmt
Simplify return