Final stage of part 2 - chat working