Remove embeds and attempt to print output probs
Merge pull request #15 from ianks/clean-up

Remove embeds and attempt to print output probs