add plotting functionality
use sns to prettify plots