Primary Dx for psychiatric rehabilitation treatment plan