Moto G first gen: Restart problem after lollipop update

After upgrading my moto g (1st gen), whenever I lock my phone it continuously starts rebooting the phone. Currently the only option I have is to keep the phone unlocked and open. How to fix it?