I cant seem to figure out what is wrong with my code when it prints out I get the words all together for example:

Enter message to be encrypted: Hello World

Enter shift amount $(1-25)$: $3$

Encrypted message: KhoorZruogg

And then when I do the opposite I get this and it doesnt even have the correct letters:

Enter message to be encrypted: KhoorZruogg

Enter shift amount $(1-25)$: $23$

Encrypted message: belloqorlddd

This is my current code:

def encode$($msg$,$ shift$)$:

newmsg $=""$

for ch in msg:

if $(\text{'}$A$\text{'}<=$ ch and ch $<=\text{'}$Z$\text{'})$:

newch $=$ chr$(($ord$($ch$)-$ ord$(\text{'}$A$\text{'})+$ shift $\%26+$ ord$(\text{'}$A$\text{'})))$

newmsg $=$ newmsg $+$ newch

elif $(\text{'}$a$\text{'}<=$ ch and ch $<=\text{'}$z$\text{'})$:

newch $=$ chr$(($ord$($ch$)-$ ord$(\text{'}$a$\text{'})+$ shift$)\%26+$ ord$(\text{'}$a$\text{'}))$

newmsg $=$ newmsg $+$ newch

else:

newch $=$ ch

newmsg $+=$ newch

return newmsg

msg $=$ input$("$Enter message to be encrypted: $")$

shiftstr $=$ input$("$Enter shift amount $(1-25)$: $")$

shift $=$ int$($shiftstr$)$

print$("$Encrypted message: $"+$ encode$($msg$,$ shift$))$