Finished the first stage implementation

1. Fixed the reward error by changing 'eta'
2. Fixed the angle setting bug in the forloop