Separate the part that builds the model from main function.  Make variable names neater.
Add readme.