Move UI setup to dedicated class, Fix bug with zoom