fix image size and add stopword and simplified class