Using *args instead of agr list

Also add some error handling to user input