Rewrote input section and added image for test setup