Bug fixes (#10, #12)