Kate Beckinsale working on her flexibility at the gym