Adding docstring.

I added docstring to the problems and moved main into #[test] functions.