Fixed plotting bug
Updated styles