Added classes to architecture diagram during discussion with Mikkel