I found the fix in another thread. Here it is -
Settings --> Device Care --> Battery --> 3 dots --> Settings Optimize Settings --> Turn OFF
And it WORKED. No more timeout issues :)
Much better option which I would recommend is - Follow same steps as above. Do not turn off Optimize Settings. Instead touch on it and you will see 3 more options under Optimize Settings. Here turn off Screen timeout.
... View more