Attempt to improve code climate score