Add a simple test + add_loop function