Train function now uses previosly found faces