Improved lithium adsorption in boron- and nitrogen-substituted graphene derivatives