For trainimg my Customize the view ability - View it on GitHub