Made define_method chainable & added tests