Updated docs and added model folder