fixed shit
fixed figure generation