Added logic for submit button