Fixing random initialization

#111
Adding a couple unit tests