Made it work =)

I've added basic functionality, and more methods.
Removed test code