Replaced user_input.split()[0] with command

This was done to improve readability of the code