adding todo comment
fixed small bug