modify reward function, restrict actions for testing
add idle action