Fixed import in styles
Addind new treatment feature