POMDP mostly done