Address issues #10 and #11