Added the ideal case on to the main_function
Added the ideal case test