slight code updates