Made additions to the potential class