Code added for calling classification method