Cleaned model building ; added some informative commented code