This is already implemented in the Rakefile but the code needs to be ported over into the actual gem where necessary