added solution for the transit lab question