Add a function in tools_td.py file in sftda module to calculate the <S^2> for excited states.

Add a function in tools_td.py file in sftda module to calculate the <S^2> for excited states. #74