Water purification by improved polymeric and graphene-based membranes