Added error handling
Added colours