Slight update to writing a wrapper gem