Adding some test example to see if I get it

Added some test cases so you can evaluate if i'm on the right path.
Sorry for multiple commits I got confused a little.