Switch to YARD <3
Adding features and stuff