Add training repo diagram, fix title