Improved examples in documentation

I improved examples in the documentation by standardizing imports, making variable names more obvious, and steps of the code easily understandable. Example `demo_elementwise.py` also used a deprecated parameter and did not work.