Made chat slightly prettier, adding test cases