Add pretty printing
Add colors to type annotations