todo: improve styling