fixed errors, generates nicer plots now