Added a lil smthng

Added the penalty so it actually works, adjusted the sizes a bit