Added square well LazyFunction example to test class