I want to be an action hero, says Rahul Vijay Working on discipline and balance