Tests added for assembly