Implemented infinite square well and simple harmonic potential