finish out predict method
add gradient calculations to lstm