Program done but don't have function for classification yet.