Add some SFX. Refactor UI stuff into ui