add .train functions
trying tests