Added lstm.py to the repo