Added test to RSPEC, alomost work