Fix js and manually add submit class