need to fix plotting still