Made withState using variables arguments