Updated docs and added UI classes diagram