Test code Improvement
Working on the ideas for the current tasks as mentioned in EP[1] in line numbers 208 and 210. Went through the test codes, most of them are assert statements though. Need to go through the code in more detail to find the complexities and simplify it as a more efficient code.