Training functionality added