add checks
minor update