Documentation

Added proper docstrings to each class. Wrote Sampler class to enable
sampling of the virus from the simulated environment.