Almost there. Just need a little tweaking