You can surely go for shoes with heels but ensure the support to your ankles, arch and heels is right and you can easily walk in them. That said, it is advisable to go for flats as there are chances you may lose your balance and fall in heels. Also, remember your top priorities should be comfort and safety.