fix code and add button style