Initial pyplot code added