added proper validation and classification after training