Added some commented code to models