Image classification using MobileNet