Adding warning and cleaning up docs