Rewrote the word_model interface toward reinforcement learning

Rewrote the simple model implementation to conform.