Improving and Adding plots