Partially working, but the wrong model - me thinks
Fixed model associations.
Cleaned up code and minor styling.