Update custom shape docs and add case class shape