added HO wavefunctions to distribution test notebook
minor updates: code cleanup, use cy_expect instead of expect in ssesolve