updated logic
1 file changed