Starting with the melody is good. Making your way back to it...even better! Try and sing or speak the words in time as you play. This helps so much with your memorization, and you'll be internalizing multiple musical elements at once! Try a few licks to get a sense of how variations and more complex arrangements are crafted.