Added code to extract CNN features