now it works and looks nice