Please improve lab implementation