Implement logging

[indrajithi]

• Currently we use print for stdout.
• Remove print across the code base
• Use logging
• Explore how to provide different coloured logs

[Mews]

If you haven't started working on this, could I be assigned to it :)