Updated script

Fixed style and button