DONE

Made return block more efficient