Tests added. We good :smile: