Lite förbättringar :)