First commit
Basic implementation
Add Untrain method